import pandas as pd


def main():
    # 读取Excel文件
    input_file = '材料学院2021级本科生学业导师选择志愿.xlsx'  # 将'your_excel_file.xlsx'替换为您的Excel文件路径
    df = pd.read_excel(input_file)

    # 创建新的DataFrame用于存储拆分后的数据
    new_data = {'专业': [], '班级': [], '学号': [], '姓名': [], '导师': [], '志愿次序': []}

    # 遍历每一行，将导师一和导师二拆分成两条记录，并加入导师类型
    for index, row in df.iterrows():
        导师一 = row['导师一']
        导师二 = row['导师二']
        new_data['专业'].extend([row['专业'], row['专业']])
        new_data['班级'].extend([row['班级'], row['班级']])
        new_data['学号'].extend([row['学号'], row['学号']])
        new_data['姓名'].extend([row['姓名'], row['姓名']])
        new_data['导师'].extend([导师一, 导师二])
        new_data['志愿次序'].extend(['导师一', '导师二'])

    # 创建新的DataFrame
    new_df = pd.DataFrame(new_data)

    # 将拆分后的数据保存到新的Excel文件
    output_file = 'new材料学院2021级本科生学业导师选择志愿.xlsx'  # 可以替换为您希望保存的文件路径
    new_df.to_excel(output_file, index=False)

    print("拆分完成，并已保存到", output_file)


if __name__ == '__main__':
    main()
